import pymysql  # pip install pymysql == 1.0.2


# 链接

# sql="SELECT username FROM person WHERE money >0"
#               地址             用户名          密码      数据库名
def select(sql, param="", mode="all", size=""):
    con = pymysql.connect(host="localhost", user="root", password="123", database="管家婆记帐系统")

    cur = con.cursor()  # 创建控制台
    #      查询  要查询的内容 来自  表    条件   表达式
    if len(param) != 0:
        cur.execute(sql, param)  # 返回的信息的个数
    else:
        cur.execute(sql)
    if mode == "one":
        return cur.fetchone()  # 返回具体内容
    elif mode == "all":
        return cur.fetchall()
    elif mode == "many":
        return cur.fetchmany(size=size)
    cur.close()
    con.close()


# sql='select username  from user where username=%s'
# param="Fra"
# print(len(select(sql, param)))


def inster(sql, param):
    con = pymysql.connect(host="localhost", user="root", password="123", database="管家婆记帐系统")

    cur = con.cursor()  # 创建控制台
    #      查询  要查询的内容 来自  表    条件   表达式
    cur.execute(sql, param)
    con.commit()  # 提交数据
    cur.close()
    con.close()

# print(select(sql))
